package com.czk.algorithm.floyd;

import java.util.Arrays;

/**
 * 弗洛伊德算法
 * @Author:ChenZhangKun
 * @Date: 2020/8/16 15:05
 */
public class FloydDemo {
    public static void main(String[] args) {
        char[] vertex={'A','B','C','D','E','F','G'};
        // 创建邻接矩阵
        final int INF=65535;
        int[][] matrix=new int[vertex.length][vertex.length];
        matrix[0]=new int[]{0,5,7,INF,INF,INF,2};
        matrix[1]=new int[]{5,0,INF,9,INF,INF,3};
        matrix[2]=new int[]{7,INF,0,INF,8,INF,INF};
        matrix[3]=new int[]{INF,9,INF,0,INF,4,INF};
        matrix[4]=new int[]{INF,INF,8,INF,0,5,4};
        matrix[5]=new int[]{INF,INF,INF,4,5,0,6};
        matrix[6]=new int[]{2,3,INF,INF,4,6,0};
        Graph graph=new Graph(vertex.length,matrix,vertex);
        graph.floyd();
        graph.showDis();
        System.out.println("============");

    }
}

/**
 * 创建图
 */
class Graph{
    /**
     * 顶点数组
     */
    private char[] vertex;
    /**
     * 保存从各个顶点出发到其他顶点的距离
     */
    private int[][] dis;
    /**
     * 保存到达目标顶点的前驱节点
     */
    private int[][] pre;

    public Graph(int length,int[][] matrix,char[] vertex) {
        this.vertex=vertex;
        this.dis=matrix;
        this.pre=new int[length][length];
        // 对pre进行初始化，存放的是前驱节点的下标
        for (int i = 0; i < length; i++) {
            Arrays.fill(pre[i],i);
        }
    }
    public void showPre(){
        for (int[] ints : pre) {
            System.out.println(Arrays.toString(ints));
        }
    }
    public void showDis(){
        for (int[] di : dis) {
            System.out.println(Arrays.toString(di));
        }
    }

    /**
     * 弗洛伊德算法
     */
    public void floyd(){
        // 临时变量
        int len=0;
        // 对中间顶点进行遍历
        for (int k = 0; k < vertex.length; k++) {
            // 从i顶点开始出发
            for (int i = 0; i < vertex.length; i++) {
                for (int j = 0; j < vertex.length; j++) {
                    // 从i顶点出发经过k到达终点j的距离
                    len=dis[i][k]+dis[k][j];
                    // 如果距离小于直连的距离
                    if (len<dis[i][j]){
                        // 更新距离
                        dis[i][j]=len;
                        // 更新前驱顶点
                        pre[i][j]=pre[k][j];
                    }
                }
            }
        }
    }
}